package com.atguigu.tingshu.common.login;

import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.execption.GuiguException;
import com.atguigu.tingshu.common.result.ResultCodeEnum;
import com.atguigu.tingshu.common.util.AuthContextHolder;
import com.atguigu.tingshu.model.user.UserInfo;
import jakarta.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.UUID;

/**
 * @author atguigu-mqx
 * @ClassName GuiGuLoginAspect
 * @description: TODO
 * @date 2023年11月27日
 * @version: 1.0
 */
@Aspect
@Component
public class GuiGuLoginAspect {

    @Autowired
    private RedisTemplate redisTemplate;
    //  @annotation(org.springframework.transaction.annotation.Transactional)

    /**
     * 环绕通知
     *
     * @param point      核心对象
     * @param guiGuLogin 注解
     * @return
     * @throws Throwable
     */
    @Around("execution(* com.atguigu.tingshu.*.api.*.*(..)) && @annotation(guiGuLogin)")
    public Object guiGuLoginAspect(ProceedingJoinPoint point, GuiGuLogin guiGuLogin) throws Throwable {
        //  判断用户是否登录? 登录成功之后，我们会生成一个token 存储到header中。
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) requestAttributes;
        //  获取到了request 请求
        HttpServletRequest request = servletRequestAttributes.getRequest();
        String token = request.getHeader("token");
        //  注解有个属性：required 表示是否需要登录
        if (guiGuLogin.required()){
            //  访问的控制器必须要登录
            //  判断token 是否已经登录。
            if (!StringUtils.isEmpty(token)){
                //  有token set key value; key=user:login:token value=userInfo;
                String userKey = RedisConstant.USER_LOGIN_KEY_PREFIX + token;
                //  从缓存中获取数据
                UserInfo userInfo = (UserInfo) redisTemplate.opsForValue().get(userKey);
                //  判断
                if (null == userInfo){
                    //  说明登录已过期。抛出异常提示信息。
                    throw new GuiguException(ResultCodeEnum.LOGIN_AUTH);
                }
            } else {
                //  说明未登录。抛出异常提示信息。
                throw new GuiguException(ResultCodeEnum.LOGIN_AUTH);
            }
        }

        //  required=false: 说明当前访问的这个控制器不需要登录.
        if (!StringUtils.isEmpty(token)){
            //  有token set key value; key=user:login:token value=userInfo;
            String userKey = RedisConstant.USER_LOGIN_KEY_PREFIX + token;
            //  从缓存中获取数据
            UserInfo userInfo = (UserInfo) redisTemplate.opsForValue().get(userKey);
            //  为了防止空指针
            if (null!=userInfo){
                //  将用户信息存储到本地
                AuthContextHolder.setUserId(userInfo.getId());
                AuthContextHolder.setUsername(userInfo.getNickname());
            }
        }
        //  proceed() 表示 去执行被这个注解修饰的方法体
        return point.proceed();
    }

    //    public void login(){
    //        //  模拟一个登录方法.
    //        //  setHeader("token", UUID.randomUUID().toString().replace("-",""));
    //        //  还需要将用户信息存储到缓存中. key-value key=token 组成
    //    }

}
